../../../../models/demos/wormhole/vit/tests/test_ttnn_optimized_sharded_vit_wh.py